tensor([[[[[0.2900, 0.3000],
           [0.3000, 0.2900]],

          [[0.3100, 0.3200],
           [0.3200, 0.3100]]],


         [[[0.3100, 0.3200],
           [0.3200, 0.3100]],

          [[0.0000, 0.0000],
           [0.0000, 0.0200]]]],



        [[[[0.3900, 0.4000],
           [0.4000, 0.3900]],

          [[0.4200, 0.4200],
           [0.4200, 0.4200]]],


         [[[0.4200, 0.4200],
           [0.4200, 0.4200]],

          [[0.0300, 0.0800],
           [0.0200, 0.0500]]]],



        [[[[0.2400, 0.2400],
           [0.2400, 0.2400]],

          [[0.2200, 0.2200],
           [0.2200, 0.2200]]],


         [[[0.2200, 0.2200],
           [0.2200, 0.2200]],

          [[0.5700, 0.5600],
           [0.5500, 0.5400]]]],



        [[[[0.0700, 0.0700],
           [0.0700, 0.0700]],

          [[0.0500, 0.0400],
           [0.0400, 0.0500]]],


         [[[0.0500, 0.0400],
           [0.0400, 0.0500]],

          [[0.3900, 0.3700],
           [0.4300, 0.3900]]]]], grad_fn=<RoundBackward1>)